package com.nineclock.common.filter;

import com.nineclock.common.entity.UserInfo;
import com.nineclock.common.threadlocal.CurrentUserHolder;
import com.nineclock.common.utils.JsonUtils;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.web.filter.GenericFilterBean;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.annotation.WebFilter;
import java.io.IOException;

/**
 * @author 黑马程序员
 * @Company http://www.ithiema.com
 * @Version 1.0
 */
@WebFilter("/*")
public class CurrentUserFilter extends GenericFilterBean {

    /**
     * 把 userInfo信息存储到当前线程中
     *
     * 在资源服务器中，验证token没有问题，会自动把 用户认证的信息存储到 session中(Security上下文中也有用户认证信息)
     * @param request
     * @param response
     * @param chain
     * @throws IOException
     * @throws ServletException
     */
    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        //获取 Security上下文对象
        SecurityContext context = SecurityContextHolder.getContext();
        //获取认证信息
        Authentication authentication = context.getAuthentication();
        //获取 认证信息中的 json字符串(如果token有问题，或者没有token，这里获取的是 anonymousUser(匿名用户))
        String jsonStr = authentication.getName();
        //如果是null，或者是匿名用户， 说明没有登录，没有登录就没有用户信息
        if(jsonStr != null && !"anonymousUser".equals(jsonStr)){
            //把json转换为 bean对象，存储到当前线程
            UserInfo userInfo = JsonUtils.toBean(jsonStr, UserInfo.class);
            CurrentUserHolder.set(userInfo);
        }
        //放行
        chain.doFilter(request, response);

    }
}
